#!/bin/bash
#SBATCH -J test
#SBATCH -p Hygon_Z100
#SBATCH -N 1
#SBATCH -n 32
#SBATCH --gres=dcu:1
module purge
module add compiler/gcc/7.3.1 mpi/openmpi/gnu/4.0.3
module add compiler/dtk/23.04.1
source $HOME/miniconda3/etc/profile.d/conda.sh
conda activate mmdet
python setup.py develop

export MIOPEN_DEBUG_DISABLE_FIND_DB=1
export MIOPEN_DEBUG_CONV_WINOGRAD=0 
export MIOPEN_DEBUG_CONV_IMPLICIT_GEMM=0

python tools/train.py configs/yolo/yolov3_dronevehiclenight_IAT_lol_15e.py